package com.briup.pts.config.auth;

import java.util.HashMap;
import java.util.Map;

/**
 * 用于维护用户与token数据的关系，
 * 在登录校验时使用
 */
public class TokenMap {
    private static final Map<String, String> MAP = new HashMap<>();

    // 存储username和token的关系
    public static void setToken(String username, String token) {
        MAP.put(username, token);
    }

    // 通过用户名获取token信息，没有则返回null
    public static String getToken(String username) {
        return MAP.get(username);
    }

    // 当退出登录时，需要将tokenMap中对应用户的token移除
    public static void remove(String username) {
        MAP.remove(username);
    }
}
